package com.atguigu.gulimall.ssoserver.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletResponse;
import java.util.UUID;

/**
 * @Author FangDaDa
 * @Description TODO
 * @Date 22:29 2021/8/16
 * @Param
 * @return
 **/

@Controller
public class LoginController {

    @Autowired
    StringRedisTemplate redisTemplate;

    @ResponseBody
    @GetMapping("/userinfo")
    public String userinfo(@RequestParam(value = "token") String token) {
        // 得到redis中的存储过的user信息，返回给子系统的session中
        // 拿着其他域名转发过来的token去redis里查
        String s = redisTemplate.opsForValue().get(token);

        return s;

    }


    // 子系统都来这
    @GetMapping("/login.html")
    public String loginPage(@RequestParam("redirect_url") String url, Model model,
                            @CookieValue(value = "sso_token", required = false) String sso_token) {
        // 非空代表就登录过了
        if (!StringUtils.isEmpty(sso_token)) {
            // 告诉子系统他的redisKey，拿着该token就可以查redis了
            return "redirect:" + url + "?token=" + sso_token;
        }
        // 子系统都没登录过才去登录页
        model.addAttribute("url", url);
        return "login";
    }

    // server端统一认证
    @PostMapping(value = "/doLogin")
    public String doLogin(@RequestParam("username") String username, @RequestParam("password") String password,
                          @RequestParam("redirect_url") String url, HttpServletResponse response) {
        // 确认用户后，生成cookie、redis中存储
        if (!StringUtils.isEmpty(username) && !StringUtils.isEmpty(password)) {
            // 登录成功跳转 跳回之前的页面
            String uuid = UUID.randomUUID().toString().replace("_", "");
            // 存储cookie， 是在server.com域名下存  redis中存储
            redisTemplate.opsForValue().set(uuid, username);

            // 需要把sso_token返回回去，给当前系统留一个记号表示我已经登录过了。设置cookie，多系统单点登录的核心要素
            Cookie sso_token = new Cookie("sso_token", uuid);
            // user中存储的url  重定向时候带着token
            response.addCookie(sso_token);

            return "redirect:" + url + "?token=" + uuid;
        }
        // 登录失败返回登录页
        return "login";
    }

}
